#!/bin/bash
# 确保脚本执行权限：chmod +x experiments/run_all_mnist_experiments.sh

# 定义实验命令列表
EXPERIMENTS=(
    # 1. MultiScaleCNN 5轮重复实验
    "RUN_ENV=remote python experiments/train_mnist.py --model multi_scale_cnn --runs 5"

    # 2. SimpleCNN 5轮重复实验
    "RUN_ENV=remote python experiments/train_mnist.py --model simple_cnn --runs 5"

    # 3. LeNet5 5轮重复实验
    "RUN_ENV=remote python experiments/train_mnist.py --model lenet5 --runs 5"

    # 4. 消融实验（移除7×7分支）
    "RUN_ENV=remote python experiments/train_mnist.py --model multi_scale_cnn --use_branch3 false --runs 5"
)

# 循环执行所有实验
for exp in "${EXPERIMENTS[@]}"; do
    echo "=================================================="
    echo "开始执行实验: $exp"
    echo "开始时间: $(date)"
    echo "=================================================="

    # 执行实验命令
    eval $exp

    # 检查上一个命令是否成功执行
    if [ $? -ne 0 ]; then
        echo "实验执行失败: $exp"
        exit 1
    fi

    echo "=================================================="
    echo "实验完成: $exp"
    echo "完成时间: $(date)"
    echo "=================================================="
    echo ""
done

echo "所有实验已全部完成！"